package com.leetcode.hash.demo5;

import java.util.HashMap;
import java.util.Map;

/**
 * @Author Zan
 * @Create 2024/5/22 11:18
 * @ClassName: Test
 * @Description : https://leetcode.cn/problems/4sum-ii/  四数相加 II
 */
public class Test {
    public static void main(String[] args) {

    }
}

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        int res = 0;
        // 先获取前两个数组的所有和的值存到集合中
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums1.length; i++) {
            for (int j = 0; j < nums2.length; j++) {
                map.put(nums1[i] + nums2[j], map.getOrDefault(nums1[i] + nums2[j], 0) + 1);
            }
        }

        // 然后获取后两个数组的和，判断是否符合条件，进行++
        for (int i = 0; i < nums3.length; i++) {
            for (int j = 0; j < nums4.length; j++) {
                res = res + map.getOrDefault(0 - nums3[i] - nums4[j], 0);
            }
        }
        return res;
    }
}